# python3 依赖环境配置
sudo apt-get update
sudo apt-get upgrade -y
sudo apt-get install -y git cmake python3-dev
sudo apt-get install -y libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev
sudo apt-get install -y python3-pip
sudo pip3 install -U pip testresources setuptools
# pytorch 依赖环境配置
sudo apt-get install -y libopenblas-base libopenmpi-dev
sudo pip3 install mpi4py
sudo pip3 install Cython
sudo pip3 install torch-1.5.0-cp36-cp36m-linux_aarch64.whl
# torchvision 配置
sudo apt-get install -y  libjpeg-dev zlib1g-dev
git clone --branch  v0.6.0 https://gitee.com/mirrors/vision torchvision
cd torchvision
sudo python3 setup.py install
